package com.ywyself.model;

import javax.sql.DataSource;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

/**
 * 在数据库表有任何变动时，运行一下 main 方法，极速响应变化进行代码重构
 * 
 * @author YuWenYue
 * @date 2017年10月24日 下午7:36:56
 */
public class _JFinalDemoGenerator {

    private static DruidPlugin createDruidPlugin() {
        // 数据库配置文件
        PropKit.use("config.txt");
        String url = PropKit.get("datasource.url");
        String user = PropKit.get("datasource.user");
        String password = PropKit.get("datasource.password");
        return new DruidPlugin(url, user, password.trim());
    }

    public static DataSource getDataSource() {
        DruidPlugin druidPlugin = _JFinalDemoGenerator.createDruidPlugin();
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }

    public static void main(String[] args) {
        // base model 所使用的包名
        String baseModelPackageName = "com.ywyself.model.base";
        // base model 文件保存路径
        String baseModelOutputDir = PathKit.getWebRootPath() + "/../src/com/ywyself/model/base";

        // model 所使用的包名 (MappingKit 默认使用的包名)
        String modelPackageName = "com.ywyself.model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";

        // 创建生成器
        Generator generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName,
                modelOutputDir);
        // 添加不需要生成的表名
        generator.addExcludedTable("ys_article");
        generator.addExcludedTable("ys_catelog");
        generator.addExcludedTable("ys_mapping");
        generator.addExcludedTable("ys_permission");
        generator.addExcludedTable("ys_role");
        generator.addExcludedTable("ys_role_permission");
        generator.addExcludedTable("ys_user");
        generator.addExcludedTable("ys_option");
        generator.addExcludedTable("ys_diary");
        generator.setMappingKitClassName("_Mapping");
        // 设置是否在 Model 中生成 dao 对象
        generator.setGenerateDaoInModel(true);
        // 设置是否生成链式 setter 方法
        generator.setGenerateChainSetter(true);
        // 设置是否生成字典文件
        generator.setGenerateDataDictionary(false);
        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为
        // "User"而非 OscUser
        generator.setRemovedTableNamePrefixes("ys_");
        // 生成
        generator.generate();
    }

}
